package cn.hanfeng.taskservice.config;

import cn.hanfeng.taskservice.runnable.SchedulingRunnable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.config.CronTask;
import org.springframework.scheduling.config.ScheduledTask;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;
import org.springframework.scheduling.config.Task;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;

/**
 * 定时任务注册器
 * <pre>
 *  定时任务使用示例
 *  1.启动类增加注解 @EnableScheduling
 *  2.相应类声明为服务 @Service
 *  3.方法上面增加 @Scheduled, 指定不同的参数以不同的方式运行定时任务
 *  注：
 *  {@code @Scheduled}中参数解释：
 *      fixedRate：表示以固定的时间间隔执行（上次开始执行和本次开始执行之间的时间间隔），不关注被执行方法实际的执行时间
 *      fixedDelay：表示以固定的延迟执行（上次执行结束和本次开始执行的时间间隔），受被执行方法执行时间的影响
 *      cron="2 * * * * *": cron表达式配置执行方法。总共6位，分别代表 秒 分 时 日 月 周
 *  spring底层执行器：（默认使用的是单线程线程池）
 *      this.localExecutor = Executors.newSingleThreadScheduledExecutor();
 *      this.taskScheduler = new ConcurrentTaskScheduler(this.localExecutor);
 *  加载流程：
 *      ScheduledAnnotationBeanPostProcessor -> ScheduledTaskRegistrar -> afterPropertiesSet()
 *  {@code @Scheduled} 的方法必须是空返回值和无参方法，直接调用这个方法是同步的
 *  </pre>
 *
 * @author hanfeng 2020/8/26 9:50
 * @version 0.0.1
 */
@Component
public class MyScheduledTaskRegistrar {

    private final Logger LOGGER = LoggerFactory.getLogger(getClass());

    @Autowired
    private ScheduledTaskRegistrar registrar;

    private final ConcurrentHashMap<String, ScheduledTask> SCHEDULED_FUTURES_MAP = new ConcurrentHashMap<>();

    /**
     * 添加任务
     *
     * @param task           任务
     * @param cronExpression cron表达式
     */
    public void addCronTask(SchedulingRunnable task, String cronExpression) {
        String taskSign = task.getTaskSign();
        if (SCHEDULED_FUTURES_MAP.containsKey(taskSign)) {
            LOGGER.error("任务[{}]已经被添加, 不能重复添加", taskSign);
            throw new RuntimeException("任务[" + taskSign + "]已经被添加, 不能重复添加");
        }
        CronTask cronTask = new CronTask(task, cronExpression);
        registrar.addCronTask(cronTask);
        ScheduledTask scheduledTask = registrar.scheduleCronTask(cronTask);
        SCHEDULED_FUTURES_MAP.put(taskSign, scheduledTask);
        LOGGER.info("任务[{}]添加成功", taskSign);
    }

    /**
     * 编辑cron任务
     * @param task
     * @param cronExpression
     */
    public void editCronTask(SchedulingRunnable task, String cronExpression) {
        String taskSign = task.getTaskSign();
        this.removeCronTask(taskSign);
        this.addCronTask(task, cronExpression);
    }

    /**
     * 移除任务
     *
     * @param taskSign 任务标识
     */
    public void removeCronTask(String taskSign) {
        if (!SCHEDULED_FUTURES_MAP.containsKey(taskSign)) {
            LOGGER.error("任务[{}]不存在, 未执行移除", taskSign);
            throw new RuntimeException("任务[" + taskSign + "]不存在, 未执行移除");
        }
        ScheduledTask scheduledTask = SCHEDULED_FUTURES_MAP.get(taskSign);
        scheduledTask.cancel();

        SCHEDULED_FUTURES_MAP.remove(taskSign);
        LOGGER.info("任务[{}]移除成功", taskSign);
    }
}
